univariate linear regression in pytorch